# Copyright (c) 2024-present AI-Labs

import os


"""
从魔搭平台下载模型权重
"""
def download_model(model_name):
    from modelscope.hub.snapshot_download import snapshot_download
    os.system(f'rm -rf checkpoints/${model_name}')
    snapshot_download(model_name, cache_dir="checkpoints/")


if __name__ == '__main__':
    download_model("ZhipuAI/glm-4-9b-chat")
    download_model("wuxlabs/Leffa")
    pass
